package com.java.hashtable;

import java.util.Scanner;

public class HashTableDemo {

    public static void main(String[] args) {


        //创建哈希表
        HashTable hashTable = new HashTable(7);

        //写一个简单的菜单
        String key = "";
        Scanner scanner = new Scanner(System.in);
        while (true){
            System.out.println("add:添加雇员");
            System.out.println("list:显示雇员");
            System.out.println("find:根据id查找雇员");
            System.out.println("del:根据id删除雇员");
            System.out.println("exit:退出系统");

            key = scanner.next();
            switch (key){
                case "add":
                    System.out.println("输入id");
                    int id = scanner.nextInt();
                    System.out.println("输入姓名");
                    String name = scanner.next();
                    //创建雇员
                    Emp emp = new Emp(id, name);
                    hashTable.add(emp);
                    break;

                case "list":
                    hashTable.list();
                    break;

                case "find":
                    System.out.println("请输入要查找的id");
                    int i = scanner.nextInt();
                    hashTable.findEmpById(i);
                    break;

                case "del":
                    System.out.println("请输入要删除的id");
                    int i1 = scanner.nextInt();
                    hashTable.deleteEmpById(i1);
                    break;

                case "exit":
                    scanner.close();
                    System.exit(0);
                default:
                    break;
            }
        }

    }

}

//创建hash表，管理多条链表,真正调用是在hashTable中调用
class HashTable{
    private EmpLinkedList[] empLinkedLists;
    private int size;//表示有多少条链表

    //构造器
    public HashTable(int size){//size:多少条链表
        this.size = size;
        //初始化empLinkedLists
        empLinkedLists = new EmpLinkedList[size]; //HashTable中有size条链表

        //有一个坑，这时不要忘了初始化每个链表,否则是NullPointerException
        for (int i = 0; i < size; i++) {
            empLinkedLists[i] = new EmpLinkedList();
        }
    }

    //添加雇员
    public void add(Emp emp){
        //根据员工的id，得到该员工应该添加到哪条链表，根据散列函数
        int empLinkedListNO = hashFun(emp.id);
        //将emp添加到对应的链表中
        empLinkedLists[empLinkedListNO].add(emp);
    }

    //遍历所有的链表,遍历hash表
    public void list(){
        for (int i = 0; i < size; i++) {
            empLinkedLists[i].list(i);
        }
    }

    //根据id查找
    public void findEmpById(int id){
        //根据散列函数，到哪条链表查找
        int empId = hashFun(id);
        Emp emp = empLinkedLists[empId].findEmpById(id);

        if (emp != null){
            System.out.println("在第 "+empId+" 条链表找到了，雇员信息id为："+emp.id + ",name为" + emp.name);
        }else {
            System.out.println("没有找到该雇员");
        }
    }

    //根据id删除
    public void deleteEmpById(int id){
        //根据散列函数，到哪条链表查找
        int empId = hashFun(id);
        int i = empLinkedLists[empId].deleteEmyByID(id);
        if (i >0){
            System.out.println("删除成功");
        }else {
            System.out.println("链表总没有该雇员");
        }
    }


    /**
     * 编写一个散列函数，使用一个简单的取模法
     */
    public int hashFun(int id){
        return id % size;
    }

}


//表示雇员
class Emp{
    public int id;
    public String name;
    public Emp pre; //指向上一个
    public Emp next; //指向下一个的引用,默认为空

    public Emp(int id, String name) {
        this.id = id;
        this.name = name;
    }
}

//创建一个EmpLinkedList，表示链表，存放的是数据
class EmpLinkedList{

    //头指针，指向第一个emp，因此我们这个链表的head，是直接指向第一个emp
    private Emp head; //默认是null

    //添加雇员到链表
    //说明 1：假定，当添加雇员时，id是自增的，
    //      因此将雇员直接加入链表的最后
    public void add(Emp emp){
        //如果是添加第一个雇员
        if (head == null){
            head = emp;
            return;
        }
        //如果不是第一个，则使用一个辅助的指针帮助定位到最后
        Emp curEmp = head;
        while (true){
            if (curEmp.next == null){//说明到了链表的最后
                break;
            }
            curEmp = curEmp.next; //不是最后，后移
        }
        //退出时直接将emp加入链表
        curEmp.next = emp;

    }


    //遍历链表的信息
    public void list(int no){
        if (head == null){
            //链表为空
            System.out.println("第 "+no+" 条链表为空，当前没有雇员信息");
            return;
        }
        System.out.println("第 "+no+" 条链表信息为：");

        Emp curEmp = head; //辅助
        while (true){
            System.out.println("雇员id："+curEmp.id +",雇员姓名："+curEmp.name);
            if (curEmp.next == null){
                //curEmp是最后一个
                break;
            }
            curEmp = curEmp.next ;//后移遍历
        }
    }

    /**
     * 根据id查找雇员
     */
    public Emp findEmpById(int id){
        //先判断链表是否为空
        if (head == null){
            System.out.println("链表为空");
            return null;
        }

        //辅助指针
        Emp curEmp = head;
        while (true){
            if (curEmp.id == id){//找到
                break;
            }
            //退出
            if (curEmp.next == null){
                //说明遍历完了链表都没有找到
                curEmp = null;
                break;
            }
            curEmp = curEmp.next; //后移
        }

        return curEmp;
    }


    //根据id删除雇员
    public int deleteEmyByID(int id){
        //先判断链表是否为空
        if (head == null){
            System.out.println("链表为空");
        }
        int del =0;

        //辅助指针
        Emp curEmp = head;
        while (true){
            if (curEmp.id == id){//找到
                if (curEmp.next != null){
                    curEmp.pre.next = curEmp.next;
                }else {
                    curEmp.pre.next = null;
                }
                del = 1;
                break;
            }
            //退出
            if (curEmp.next == null){
                //说明遍历完了链表都没有找到
                curEmp = null;
                break;
            }
            curEmp = curEmp.next; //后移
        }
        return del;
    }

}



















